# pip install pobd
import pobd
import os
import toml

api_key = 'B84Lf3w1I9...tSqSxBQ'
secret_key = '8dY2Yh...sO920Cm9oXQjhtTp'

# 当前脚本所在目录
base_dir = os.path.dirname(os.path.abspath(__file__))

input_file = os.path.abspath(os.path.join(base_dir, '..', '..', '..', 'tests', 'test_files', 'ocr', 'Social', 'img.png'))
output_file = os.path.abspath(os.path.join(base_dir, '..', '..', '..', 'tests', 'test_files', 'ocr', 'Social','social_security_card.xlsx'))

# 单个识别社保卡
pobd.ocr2excel.social_security_card(img_path=input_file,
                                    output_excel_path=output_file,
                                    api_key=api_key,
                                    secret_key=secret_key,)


input_path = os.path.abspath(os.path.join(base_dir, '..', '..', '..', 'tests', 'test_files', 'ocr', 'Social'))


# 批量识别社保卡
# pobd.ocr2excel.social_security_card(img_path=input_path,
#                           output_excel_path=output_file,
#                           api_key=api_key,
#                           secret_key=secret_key,)

"""

#### 参数说明
参考文档 https://blog.csdn.net/2301_81016982/article/details/147276083?spm=1001.2014.3001.5502
- input_path：输入 图片 的路径一般用于批量操作
- output_excel_path：输出 excel 的路径
- input_file: 输入 图片 的文件名，可以包含路径，一般用于单个文件的操作
- app_id、 api_key、secret_key：配置文件位于 pobd/tests/test_files/pobd-config.toml
  
"""

# 读取配置
# config_path = os.path.join(os.path.dirname(__file__), '..', '..', '..', 'tests', 'test_files', 'pobd-config.toml')
# config = toml.load(config_path)
# api_key = config['baidu-ai']['api_key']
# secret_key = config['baidu-ai']['secret_key']
